"""
Copyright (c) 2022 Huawei Technologies Co.,Ltd.

openGauss is licensed under Mulan PSL v2.
You can use this software according to the terms and conditions of the Mulan PSL v2.
You may obtain a copy of Mulan PSL v2 at:

          http://license.coscl.org.cn/MulanPSL2

THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
See the Mulan PSL v2 for more details.
"""
"""
Case Type   : 故障&可靠性测试
Case Name   : 延迟等待过程中kill -3 各节点进程
Create At   : 2021/03/11
@zou_jialiang0505328126
Description :
    1.创建表
    2.切换备节点为级联备
    3.设置延迟为1min
    4.插入数据
    5.5s/60s查询各节点
    6.设置延迟为1min
    7.更新数据
    8.60s/300s查询各节点
    9.设置延迟为2min
    10.删除数据
    11.60s/120s查询各节点
    12.重复执行4-11 20次
Expect      :
    1.建表成功
    2.设置成功
    3.切换成功
    4.插入成功
    5.60s才同步
    6.设置成功
    7.更新成功
    8.300s才同步
    9.设置成功
    10.删除成功
    11.120s后才同步
    12.通过率100%
History     :
    2021/7/5 n@ningyali 11-12步骤之间增加步骤：等待同步完成
"""

import unittest
import time
from yat.test import Node
from testcase.utils.Common import Common
from testcase.utils.CommonSH import CommonSH
from testcase.utils.CommonSH import RestartDbCluster
from testcase.utils.Constant import Constant
from testcase.utils.Logger import Logger


class RecoveryDelay(unittest.TestCase):
    commonshpri = CommonSH('PrimaryDbUser')
    nodes_tuple = ('PrimaryDbUser', 'Standby1DbUser', 'Standby2DbUser')

    @RestartDbCluster(*nodes_tuple)
    def setUp(self):
        self.log = Logger()
        self.log.info('-----Opengauss_Reliability_Dbsys_Case266开始执行------')
        self.db_primary_root_node = Node(node='PrimaryRoot')
        self.db_standby_node = \
            [Node(node='Standby1Root'), Node(node='Standby2Root')]
        self.comshsta = \
            [CommonSH('Standby1DbUser'), CommonSH('Standby2DbUser')]
        self.constant = Constant()
        self.com_pri = Common('PrimaryRoot')
        self.com_sta = [Common('Standby1Root'), Common('Standby2Root')]
        self.tb_name = 'tb_256'

        self.log.info('=====安装ntp服务=======')
        install_cmd = "yum install -y ntp && " \
                      "systemctl enable ntpd && systemctl start ntpd"
        for i in range(2):
            result = self.db_standby_node[i].sh(install_cmd).result()
            self.log.info(result)

        self.log.info('------同步集群时间--------')
        for i in range(2):
            current = self.db_primary_root_node.sh(
                "date \"+%m/%d/%Y %H:%M:%S\"").result()
            self.log.info(current)
            datecmd = f'date -s "{current}";hwclock --systohc;' \
                f'service ntpd stop;ntpdate ntp.api.bz;service ntpd start'
            self.log.info(datecmd)
            result = self.db_standby_node[i].sh(datecmd).result()
            self.log.info(result)

        self.log.info('--------等待主备一致------------')
        for i in range(2):
            result = self.comshsta[i].check_data_consistency()
            self.assertTrue(result)

    def test_recoverydelay(self):
        self.log.info('-------创建表------')
        sql = f"drop table if exists {self.tb_name};" \
            f"create table {self.tb_name}(i int, s char(10));"
        result = self.commonshpri.executDbSql(sql)
        self.log.info(result)
        self.assertIn(self.constant.TABLE_CREATE_SUCCESS, result)

        self.log.info('-------将备节点切换为级联备---------')
        result = self.comshsta[0].buildStandby('-M cascade_standby')
        self.assertIn(self.constant.REBUILD_SUCCESS_MSG, result)
        result = self.comshsta[0].exec_refresh_conf()
        self.assertTrue(result)

        self.log.info('--------等待主备一致------------')
        result = self.comshsta[1].check_data_consistency()
        self.assertTrue(result)
        for i in range(90):
            result = self.commonshpri.check_cascade_standby_consistency()
            if result:
                break
            time.sleep(20)
        self.assertTrue(result)

        for loop_i in range(5):
            self.log.info('-----设置recovery_min_apply_delay=1min----')
            result = self.commonshpri.executeGsguc(
                'reload', self.constant.GSGUC_SUCCESS_MSG,
                'recovery_min_apply_delay=1min')
            self.assertTrue(result)

            self.log.info('-------插入数据------')
            sql = f"insert into {self.tb_name} values (1, 'test');"
            result = self.commonshpri.executDbSql(sql)
            self.log.info(result)
            self.assertIn('INSERT', result)

            self.log.info('----5s/60s查询各节点--------')
            sql = f"select sysdate;select * from {self.tb_name};"
            time.sleep(5)
            for i in range(2):
                result = self.comshsta[i].executDbSql(sql)
                self.log.info(result)
                self.assertIn('(0 rows)', result)
            time.sleep(70)
            for i in range(2):
                self.log.info(f'-----------round {i}-------------')
                result = self.comshsta[i].executDbSql(sql)
                self.log.info(result)
                self.assertIn('1 | test', result)

            self.log.info('-----设置recovery_min_apply_delay=5min----')
            result = self.commonshpri.executeGsguc(
                'reload', self.constant.GSGUC_SUCCESS_MSG,
                'recovery_min_apply_delay=5min')
            self.assertTrue(result)

            self.log.info('-------更新数据------')
            sql = f"update {self.tb_name} set i=5;"
            result = self.commonshpri.executDbSql(sql)
            self.log.info(result)
            self.assertIn('UPDATE', result)

            self.log.info('----60s/300s查询各节点--------')
            sql = f"select sysdate;select * from {self.tb_name};"
            time.sleep(70)
            for i in range(2):
                result = self.comshsta[i].executDbSql(sql)
                self.log.info(result)
                self.assertIn('1 | test', result)
            time.sleep(260)
            for i in range(2):
                result = self.comshsta[i].executDbSql(sql)
                self.log.info(result)
                self.assertIn('5 | test', result)

            self.log.info('-----设置recovery_min_apply_delay=2min----')
            result = self.commonshpri.executeGsguc(
                'reload', self.constant.GSGUC_SUCCESS_MSG,
                'recovery_min_apply_delay=2min')
            self.assertTrue(result)

            self.log.info('-------删除数据------')
            sql = f"delete from {self.tb_name};"
            result = self.commonshpri.executDbSql(sql)
            self.log.info(result)
            self.assertIn('DELETE', result)

            self.log.info('----60s查询各节点，数据未同步--------')
            sql = f"select sysdate;select * from {self.tb_name};"
            time.sleep(60)
            for i in range(2):
                result = self.comshsta[i].executDbSql(sql)
                self.log.info(result)
                self.assertIn('5 | test', result)

            time.sleep(60)

            self.log.info('--------120s后等待主备一致------------')
            result = self.comshsta[1].check_data_consistency()
            self.assertTrue(result)
            for i in range(90):
                result = self.commonshpri.check_cascade_standby_consistency()
                if result:
                    break
                time.sleep(20)
            self.assertTrue(result)

            self.log.info('----主备一致后查询各节点，数据同步--------')
            for i in range(2):
                result = self.comshsta[i].executDbSql(sql)
                self.log.info(result)
                self.assertIn('0 rows', result)

    def tearDown(self):
        self.log.info('---------环境清理----------')
        self.log.info('-------------还原recovery_min_apply_delay----------')
        result = self.commonshpri.executeGsguc(
            'set', self.constant.GSGUC_SUCCESS_MSG,
            'recovery_min_apply_delay=0')
        self.log.info(result)

        self.log.info('--------还原集群-------')
        self.comshsta[0].buildStandby('-M standby')
        result = self.comshsta[0].exec_refresh_conf()
        self.log.info(result)

        self.log.info('-----------重启数据库-----------')
        result = self.commonshpri.stopDbCluster()
        self.log.info(result)
        result = self.commonshpri.startDbCluster()
        self.log.info(result)

        time.sleep(5)
        self.log.info('--------删除表-------')
        sql = f"drop table if exists {self.tb_name};"
        result = self.commonshpri.executDbSql(sql)
        self.log.info(result)
        self.log.info('----pengauss_Reliability_Dbsys_Case266执行结束-------')
